package com.ctgu.ImageRecognition.util;

import com.ctgu.ImageRecognition.vo.Labels;
import com.ctgu.ImageRecognition.vo.ResultVO;
import com.tencentcloudapi.common.Credential;
import com.tencentcloudapi.common.exception.TencentCloudSDKException;
import com.tencentcloudapi.common.profile.ClientProfile;
import com.tencentcloudapi.common.profile.HttpProfile;
import com.tencentcloudapi.tiia.v20190529.TiiaClient;
import com.tencentcloudapi.tiia.v20190529.models.DetectLabelProRequest;
import com.tencentcloudapi.tiia.v20190529.models.DetectLabelProResponse;

public class DetectLabelProUtils {
    public static ResultVO identify(String imageUrl) {
        try{
            // 实例化一个认证对象，入参需要传入腾讯云账户 SecretId 和 SecretKey，此处还需注意密钥对的保密
            // 代码泄露可能会导致 SecretId 和 SecretKey 泄露，并威胁账号下所有资源的安全性。以下代码示例仅供参考，建议采用更安全的方式来使用密钥，请参见：https://cloud.tencent.com/document/product/1278/85305
            // 密钥可前往官网控制台 https://console.cloud.tencent.com/cam/capi 进行获取
            Credential cred = new Credential("AKIDkJbcNs63i7F6g03pKYVBaiaBVYKp30h7", "sUpuhDSR8YhaCXbOu97WkiQt63kBMJ0R");
            // 实例化一个http选项，可选的，没有特殊需求可以跳过
            HttpProfile httpProfile = new HttpProfile();
            httpProfile.setEndpoint("tiia.tencentcloudapi.com");
            // 实例化一个client选项，可选的，没有特殊需求可以跳过
            ClientProfile clientProfile = new ClientProfile();
            clientProfile.setHttpProfile(httpProfile);
            // 实例化要请求产品的client对象,clientProfile是可选的
            TiiaClient client = new TiiaClient(cred, "ap-guangzhou", clientProfile);
            // 实例化一个请求对象,每个接口都会对应一个request对象
            DetectLabelProRequest req = new DetectLabelProRequest();
            req.setImageUrl(imageUrl);
            // 返回的resp是一个DetectLabelProResponse的实例，与请求对象对应
            DetectLabelProResponse resp = client.DetectLabelPro(req);
            // 输出json格式的字符串回包
            String resultString=DetectLabelProResponse.toJsonString(resp);
            // 将json字符串解析成对象
//            Labels label=GsonUtils.fromJson(resultString);
//            System.out.println(resultString);
//            System.out.println("识别结果 ==>"+label.getName() + "\t" + label.getConfidence() +
//                    "\t" + label.getFirstCategory() + "\t" + label.getSecondCategory());

            return ResultUtil.success(resultString, resp);
        } catch (TencentCloudSDKException e) {
            System.out.println(e.toString());
            return ResultUtil.fail("识别图片失败",null);
        }
    }
}
